package cn.tedu.community.commons.excepion.handler;

import cn.tedu.community.commons.excepion.ServiceException;
import cn.tedu.community.commons.restful.JsonResult;
import cn.tedu.community.commons.restful.ServiceCode;
import lombok.extern.slf4j.Slf4j;

import org.springframework.security.access.AccessDeniedException;
import org.springframework.security.authentication.BadCredentialsException;
import org.springframework.security.authentication.DisabledException;
import org.springframework.security.authentication.InternalAuthenticationServiceException;
import org.springframework.security.core.AuthenticationException;
import org.springframework.validation.BindException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import javax.validation.ConstraintViolation;
import javax.validation.ConstraintViolationException;
import java.util.Set;

/**
 * 全局处理异常
 * 无论是哪个处理请求的过程中出现异常，每种类型的异常只需要编写一段处理异常的代码即可！
 */
@Slf4j
@RestControllerAdvice
public class GlobalExceptionHandler {

    /**
     * 处理ServiceException相关的统一异常
     */
    @ExceptionHandler({ServiceException.class})
    public JsonResult<Void> handleServiceException(ServiceException e){
        log.warn("程序过程中出现ServiceException，commons全局处理异常将统一处理！");
        log.warn("异常状态码：{}，信息：{}",e.getServiceCode().getValue(),e.getMessage());
        return JsonResult.fail(e);
    }

    /**
     * BindException
     * 用于检查controller请求方法中 [封装好] 的请求参数
     */
    @ExceptionHandler({BindException.class})
    public JsonResult<Void> handleBindException(BindException e){
        log.warn("程序过程中出现BindException，将统一处理！");
        log.warn("异常信息：{}",e.getMessage());
        return JsonResult.fail(ServiceCode.ERR_BAD_REQUEST,e.getFieldError().getDefaultMessage());
    }

    /**
     * ConstraintViolationException
     * 用于检查controller请求方法中 [未封装] 的请求参数
     */
    @ExceptionHandler
    public JsonResult<Void> handleConstraintViolationException(ConstraintViolationException e){
        log.warn("程序过程中出现ConstraintViolationException，将统一处理！");
        log.warn("异常信息：{}",e.getMessage());
        Set<ConstraintViolation<?>> constraintViolations = e.getConstraintViolations();
        String message = "";
        for(ConstraintViolation<?> constraintViolation : constraintViolations){
            message = constraintViolation.getMessage();
        }
        return JsonResult.fail(ServiceCode.ERR_BAD_REQUEST,message);
    }

    @ExceptionHandler({InternalAuthenticationServiceException.class,
            BadCredentialsException.class})
    public JsonResult<Void> handleAuthenticationException(AuthenticationException e) {
        log.warn("程序运行过程中出现AuthenticationException，将统一处理！");
        log.warn("异常信息：{}", e.getMessage());
        String message = "登录失败，用户名或密码错误！";
        return JsonResult.fail(ServiceCode.ERR_UNAUTHORIZED, message);
    }

    @ExceptionHandler
    public JsonResult<Void> handleDisabledException(DisabledException e) {
        log.warn("程序运行过程中出现DisabledException，将统一处理！");
        log.warn("异常信息：{}", e.getMessage());
        String message = "登录失败，此账号已经被禁用！";
        return JsonResult.fail(ServiceCode.ERR_UNAUTHORIZED_DISABLED, message);
    }

    /**
     * 处理管理员权限异常
     */
    @ExceptionHandler
    public JsonResult<Void> handleAccessDeniedException (AccessDeniedException e){
        log.warn("程序运行过程中出现AccessDeniedException，将统一处理！");
        log.warn("异常信息：{}", e.getMessage());
        String message = "禁止访问，您当前登录的账号无此操作权限！";
        return JsonResult.fail(ServiceCode.ERR_FORBIDDEN, message);
    }

    /**
     * Throwable
     * 用于避免因为某个异常未被处理，导致服务器端响应500错误
     */
    @ExceptionHandler
    public JsonResult<Void> handleThrowable(Throwable e){
        log.warn("程序过程中出现ThrowableException，将统一处理");
        log.warn("异常类型：{}",e.getClass());
        log.warn("异常信息：{}",e.getMessage());
        String message = "【开发阶段专用】你的服务器端出了异常，请检查服务器端控制台中的异常信息，并补充处理此类异常的方法！";
        return JsonResult.fail(ServiceCode.ERR_UNKNOWN,message);

    }
}
